package generator.main.utils;

import java.io.FileInputStream;
import java.io.InputStream;

import org.dom4j.Document;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;

/**
 * 加载配置文件的一些方法
 * @author 郭军辉
 *
 * 2017年8月11日上午12:41:01
 */
public class ConfigFileUtils {
	/***
	 * 读出需要生成表的xml配置，获取根节点
	 * 
	 * @return
	 * @throws Exception
	 */
	public static Element getConeratorConfigXMLRootElement(String path) throws Exception {
		/** 将src下面的xml转换为输入流 **/
		InputStream inputStream = new FileInputStream(path);
		/** 创建SAXReader读取器，专门用于读取xml */
		SAXReader saxReader = new SAXReader();
		/** 根据saxReader的read重写方法可知，既可以通过inputStream输入流来读取，也可以通过file对象来读取 */
		Document document = saxReader.read(inputStream);
		/** 获取xml根节点的 */
		Element root = document.getRootElement();
		if(root == null) {
			throw new RuntimeException("配置文件读取错误！");
		}else {
			return root;
		}
	}
}
